#!/usr/bin/env python3

import shelve
from datetime import datetime
from flask import Flask, request, render_template, redirect, escape, Markup

application = Flask(__name__)

DATA_FILE = 'guestbook.dat'


def save_data(name, comment, create_at):
    """
    保存提交的数据
    """
    # 通过shelve模块打开数据库文件
    database = shelve.open(DATA_FILE)

    # 如果没有数据库则创建
    if 'greeting_list' not in database:
        greeting_list = []
    else:
    # 获取数据
        greeting_list = database['greeting_list']

    # 提交数据添加到表头
    greeting_list.insert(0, {
        'name': name,
        'comment':comment,
        'create_at':create_at,
        })

    # 更新数据库
    database['greeting_list'] = greeting_list

    # 关闭数据库文件
    database.close()



def load_data():
    """
    返回已提交的数据
    """

    # 通过shelve模块打开数据库文件
    database = shelve.open(DATA_FILE)

    # 返回greeting_list.如无则返回空变
    greeting_list = database.get('greeting_list', [])
    database.close()
    return greeting_list



@application.route('/')
def index():
    """
    使用模板显示页面
    """

    # 读取已提交的数据
    greeting_list = load_data()
    return render_template('index.html', greeting_list=greeting_list)



@application.route('/post', methods=['POST'])
def post():
    """
    提交评论的URL
    """

    # 获取已提交的数据
    name = request.form.get('name')
    comment = request.form.get('comment')
    create_at = datetime.now()

    # 保存后重定向到首页
    save_data(name, comment, create_at)

    # 保存后重定向到首页
    return redirect('/')



@application.template_filter('n12br')
def n12br_filter(s):
    """
    将换行符置换为br标签的模板过滤器
    """

    return escape(s).replace('\n', Markup('<br>'))



@application.template_filter('datetime_fmt')
def datetime_fmt_filter(dt):
    """
    使datetime对象更容易分辨的模板过滤器
    """

    return dt.strftime('%Y/%m/%d %H:%M:%S')


if __name__ == '__main__':
    # 在IP地址127.0.0.1的8001端口运行应用程序
    application.run('127.0.0.1', 8001, debug=True)


